cfg = {
    # 基本设置
    "batch_size": 32,
    "input_size": (240, 320),  # (h,w)

    "bzw_detection": "bzw_detection",
    "bzw": "bzw",
    "hz": "hz",
    "jtbz": "jtbz",


    "jtbz_labels": ["停止让行", "禁止驶入", "向右转弯", "向左转弯", "禁止掉头", "禁止左拐", "车辆慢行", "右道封闭", "左道封闭"],

    # "bzw_detection_labels": ["LuDeng", "CheKu", "DaoZha", "JiaoTongDeng",
    #                      "BaoJingTai", "LED", "TFT", "LTXS", "YYBB", "WXCD", "JT"],   # xml annotation

    "bzw_detection_labels": ['no_left', 'right', 'js_50', '!_left', 'no_dt', 'no_right', 'left', 'mx'],  # xml annotation

    # "jtbz_detection_list": ['no_left', 'right', 'js_50', '!_left', 'no_dt', 'no_right', 'left', 'mx'],

    # anchor配置
    # "min_sizes": [[(9, 7), (24, 20), (39, 35)], [(54, 41), (65, 61), (81, 66)],
    #               [(94, 86), (113, 95), (131, 122)], [(137, 128), (172, 162), (176, 210)]],
    "min_sizes":[[10, 16, 24], [32, 48], [64, 96], [128, 192, 256]],
    "steps": [8, 16, 32, 64],
    "match_thresh": 0.45,
    "variances": [0.1, 0.2],
    "clip": False,


    # 推理
    "score_threshold": 0.5,   # 得分阈值
    "nms_threshold": 0.4,     # nms阈值
    "max_number_keep": 200
}
